package cn.tedu.sp11.filter;

import cn.tedu.web.util.JsonResult;
import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.exception.ZuulException;
import org.apache.commons.lang.StringUtils;
import org.springframework.cloud.netflix.zuul.filters.support.FilterConstants;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

@Component
public class AccessFilter extends ZuulFilter {

    //设置过滤器的类型  pre请求被路由之前调用，routing请求路由到微服务，post在路由到微服务以后执行,error在其他阶段发生错误时执行该过滤器
    @Override
    public String filterType() {
//        return "pre";
        return FilterConstants.PRE_TYPE;
    }

    //设置过滤器添加的位置顺序号
    /*在默认过滤器的第五个过滤器中，在上下文对象(RequestContext)中添加了serviceId,
    所以这里自定义的过滤器的顺序号必须大于5后面的判断当前请求中才能获取到请求中的serviceId
    */
    @Override
    public int filterOrder() {
        return 6;
    }

    //判断对当前请求，是否要执行过滤代码
    @Override
    public boolean shouldFilter() {
        /*
         如果调用商品，必须登录才能调用，调用其它服务不必登录
         */

        //得到当前请求的后台服务的服务id
        RequestContext context = RequestContext.getCurrentContext();
        String serviceId =(String) context.get(FilterConstants.SERVICE_ID_KEY);
        //判断服务id是否是"item-service"
        return "item-service".equalsIgnoreCase(serviceId);

    }

    //过滤代码
    @Override
    public Object run() throws ZuulException {

        //得到request对象
        RequestContext context = RequestContext.getCurrentContext();
        HttpServletRequest request = context.getRequest();
        //接收token参数
        String token = request.getParameter("token");
        //判断token是否存在
        if (StringUtils.isBlank(token)){
            //此设置会阻止请求被路由到后台微服务
            context.setSendZuulResponse(false);
            //向客户端的响应,向客户端返回响应"没有登录"
            //JsonResult{code:400, msg:"not login", data:null}
            String json = JsonResult.err().code(JsonResult.NOT_LOGIN).msg("Not login! 没有登录!").toString();
            //响应头
            context.addZuulResponseHeader("Content-Type", "application/json;charset=UTF-8");
            //响应体
            context.setResponseBody(json);
        }
        //zuul过滤器返回的数据设计为以后扩展使用，
        //目前该返回值没有被使用
        return null;
    }
}
